package com.alatus.djl.service;

import ai.djl.MalformedModelException;
import ai.djl.translate.TranslateException;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;

/**
 * @author: Alatus
 * @create: 2025-02-12 17:48
 * @description: DJL训练模型
 **/
public interface TrainService {
    /**
     *
     * @数据集的位置 datasetPath
     * @模型名字 modelName
     * @模型存放路径 modelPath
     */
    void train(String datasetPath,String modelName,String modelPath) throws IOException, TranslateException;
    /**
     *
     * @图片 图片
     */
    String predict(MultipartFile image) throws MalformedModelException, IOException, TranslateException;
}
